package com.warm.orm.eq.sys.agent;

import cn.hutool.core.util.StrUtil;
import com.easy.query.api.proxy.client.EasyEntityQuery;
import com.easy.query.api.proxy.entity.select.EntityQueryable;
import com.easy.query.core.api.pagination.EasyPageResult;
import com.easy.query.solon.annotation.Db;
import com.warm.core.page.TableDataInfo;
import com.warm.orm.domain.sys.SysDictData;
import com.warm.orm.domain.sys.proxy.SysDictDataProxy;
import com.warm.orm.utils.PageUtils;
import com.warm.system.api.orm.agent.SysDictDataAgent;
import org.noear.solon.annotation.Component;

import java.util.List;

/**
 * 字典表 数据层 orm代理类
 *
 * @author warm
 */
@Component
public class SysDictDataAgentImpl implements SysDictDataAgent {

    @Db
    private EasyEntityQuery easyEntityQuery;

    /**
     * 查询字典数据
     *
     * @param dictType 字典类型
     * @return 字典数据
     */
    @Override
    public int countDictDataByType(String dictType) {
        return this.easyEntityQuery.queryable(SysDictData.class).where(t -> t.dictType().eq(dictType)).intCount();
    }

    /**
     * 通过字典ID删除字典数据信息
     *
     * @param dictCode 字典数据ID
     * @return 结果
     */
    @Override
    public int deleteDictDataById(Long dictCode) {
        return (int) this.easyEntityQuery.deletable(SysDictData.class).where(t -> t.dictCode().eq(dictCode)).executeRows();
    }

    /**
     * 批量删除字典数据信息
     *
     * @param dictCodes 需要删除的字典数据ID
     * @return 结果
     */
    @Override
    public int deleteDictDataByIds(Long[] dictCodes) {
        return (int) this.easyEntityQuery.deletable(SysDictData.class).where(t -> t.dictCode().in(dictCodes)).executeRows();
    }

    /**
     * 新增字典数据信息
     *
     * @param dictData 字典数据信息
     * @return 结果
     */
    @Override
    public int insertDictData(SysDictData dictData) {
        return (int) this.easyEntityQuery.insertable(dictData).executeRows();
    }

    /**
     * 根据字典数据ID查询信息
     *
     * @param dictCode 字典数据ID
     * @return 字典数据
     */
    @Override
    public SysDictData selectDictDataById(Long dictCode) {
        return this.easyEntityQuery.queryable(SysDictData.class).where(t -> t.dictCode().eq(dictCode)).firstOrNull();
    }

    /**
     * 根据字典类型查询字典数据
     *
     * @param dictType 字典类型
     * @return 字典数据集合信息
     */
    @Override
    public List<SysDictData> selectDictDataByType(String dictType) {
        return this.easyEntityQuery
                .queryable(SysDictData.class)
                .where(t -> t.dictType().eq(dictType))
                .orderBy(t -> t.dictSort().asc())
                .toList();
    }

    /**
     * 根据条件分页查询字典数据
     *
     * @param dictData 字典数据信息
     * @return 字典数据集合信息
     */
    @Override
    public List<SysDictData> selectDictDataList(SysDictData dictData) {
        return this.buildQuery(dictData).orderBy(t -> t.dictSort().asc()).toList();
    }

    private EntityQueryable<SysDictDataProxy, SysDictData> buildQuery(SysDictData dictData) {
        return this.easyEntityQuery
                .queryable(SysDictData.class)
                .where(t -> {
                    t.dictType().eq(StrUtil.isNotBlank(dictData.getDictType()), dictData.getDictType());
                    t.dictLabel().like(StrUtil.isNotBlank(dictData.getDictLabel()), dictData.getDictLabel());
                    t.status().eq(StrUtil.isNotBlank(dictData.getStatus()), dictData.getStatus());
                });
    }

    /**
     * 根据条件分页查询字典数据
     *
     * @param dictData 字典数据信息
     * @return 字典数据集合信息
     */
    @Override
    public TableDataInfo<SysDictData> selectDictDataPage(SysDictData dictData) {
        PageUtils.startPage();
        EasyPageResult<SysDictData> pageResult = this.buildQuery(dictData)
                .orderBy(t -> t.dictSort().asc())
                .toPageResult(PageUtils.getLocalPage().getPageNum(), PageUtils.getLocalPage().getPageSize());
        return new TableDataInfo(pageResult.getData(), (int) pageResult.getTotal());
    }

    /**
     * 根据字典类型和字典键值查询字典数据信息
     *
     * @param dictType  字典类型
     * @param dictValue 字典键值
     * @return 字典标签
     */
    @Override
    public String selectDictLabel(String dictType, String dictValue) {
        return this.easyEntityQuery
                .queryable(SysDictData.class)
                .where(t -> {
                    t.dictType().eq(dictType);
                    t.dictValue().eq(dictValue);
                })
                .selectColumn(t -> t.dictLabel().setPropertyType(String.class))
                .firstOrNull();
    }

    /**
     * 修改字典数据信息
     *
     * @param dictData 字典数据信息
     * @return 结果
     */
    @Override
    public int updateDictData(SysDictData dictData) {
        return (int) this.easyEntityQuery.updatable(dictData).executeRows();
    }

    /**
     * 同步修改字典类型
     *
     * @param oldDictType 旧字典类型
     * @param newDictType 新旧字典类型
     * @return 结果
     */
    @Override
    public int updateDictDataType(String oldDictType, String newDictType) {
        return (int) this.easyEntityQuery
                .updatable(SysDictData.class)
                .setColumns(c -> c.dictType().set(newDictType))
                .where(t -> t.dictType().eq(oldDictType))
                .executeRows();
    }
}
